package com.heima.admin.service.impl;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.heima.admin.dto.LoginDto;
import com.heima.admin.entity.AdUser;
import com.heima.admin.mapper.AdUserMapper;
import com.heima.admin.service.IAdUserService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.heima.common.dto.ResponseResult;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.util.AppJwtUtil;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;
import org.springframework.util.StringUtils;

import java.util.HashMap;
import java.util.Map;

/**
 * <p>
 * 管理员用户信息表 服务实现类
 * </p>
 *
 * @author mcm
 * @since 2022-04-10
 */
@Service
public class AdUserServiceImpl extends ServiceImpl<AdUserMapper, AdUser> implements IAdUserService {

    @Override
    public ResponseResult login(LoginDto dto) {
        // 判断用户名和密码是否为空
        if (dto == null || StringUtils.isEmpty(dto.getName()) || StringUtils.isEmpty(dto.getPassword())) {
            return ResponseResult.errorResult(AppHttpCodeEnum.PARAM_INVALID);
        }
        // 根据用户名查询用户
        LambdaQueryWrapper<AdUser> query = new LambdaQueryWrapper<>();
        query.eq(AdUser::getName, dto.getName());
        AdUser one = this.getOne(query);
        // 不存在提示用户不存在
        if (one == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.DATA_NOT_EXIST);
        }
        // 存在的话,取出用户数据库中的salt
        String salt = one.getSalt();
        // 使用用户传入的密码+salt 进行md5 加密
        String pwd = dto.getPassword() + salt;
        String md5 = DigestUtils.md5DigestAsHex(pwd.getBytes());
        // 对比数据库中存储的密码
        // 密码不一致,提示密码错误
        if (!md5.equals(one.getPassword())) {
            return ResponseResult.errorResult(AppHttpCodeEnum.LOGIN_PASSWORD_ERROR);
        }
        // 密码一致,使用jwt生成token
        Map<String, Object> userInfo = new HashMap<>();
        userInfo.put("userId", one.getId());
        String token = AppJwtUtil.getToken(userInfo);
        // 给前端返回对应格式的数据
        AdUser user = new AdUser();
        user.setName(one.getName());
        Map<String, Object> map = new HashMap<>();
        map.put("user", user);
        map.put("token", token);
        return ResponseResult.okResult(map);
    }
}

